package com.wutu.nbyjk8.util;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

/**
 * AES 是一种可逆加密算法，对用户的敏感信息加密处理
 * 对原始数据进行AES加密后，在进行Base64编码转化；
 * 正确
 */
public class AESUtils {
    public static BASE64Encoder base64Encoder = new BASE64Encoder();
    public static BASE64Decoder base64Decoder = new BASE64Decoder();
    /**
     * 加密
     * @param sSrc        加密前json字符串
     * @param sKey        参数加密秘钥 加密用的Key 可以用26个字母和数字组成 此处使用AES-128-CBC加密模式，key需要为16位。
     * @param ivParameter 加密参数
     * @return 加密后的字符串
     */
    public static String encrypt(String sSrc, String sKey, String ivParameter) {
        return encrypt(sSrc, "utf-8", sKey, ivParameter);
    }

    /**
     * 加密
     * @param sSrc           加密前json字符串
     * @param encodingFormat 编码集
     * @param sKey           参数加密秘钥 加密用的Key 可以用26个字母和数字组成 此处使用AES-128-CBC加密模式，key需要为16位。
     * @param ivParameter    加密参数
     * @return 加密后的字符串
     */
    public static String encrypt(String sSrc, String encodingFormat, String sKey, String ivParameter) {
        try {
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
            byte[] raw = sKey.getBytes();
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
            IvParameterSpec iv = new IvParameterSpec(ivParameter.getBytes());//使用CBC模式，需要一个向量iv，可增加加密算法的强度
            cipher.init(Cipher.ENCRYPT_MODE, skeySpec, iv);
            byte[] encrypted = cipher.doFinal(sSrc.getBytes(encodingFormat));
            return new BASE64Encoder().encode(encrypted);//此处使用BASE64做转码。
        } catch (Exception ex) {
            throw new RuntimeException(ex);
        }
    }

    /**
     * 解密
     * @param sSrc        加密json字符串
     * @param sKey        参数加密秘钥 加密用的Key 可以用26个字母和数字组成 此处使用AES-128-CBC加密模式，key需要为16位。
     * @param ivParameter 加密参数
     * @return 解密后的字符串
     */
    public static String decrypt(String sSrc, String sKey, String ivParameter) {
        return decrypt(sSrc, "utf-8", sKey, ivParameter);
    }

    /**
     * 解密
     * @param sSrc           加密json字符串
     * @param encodingFormat 编码集
     * @param sKey           参数加密秘钥 加密用的Key 可以用26个字母和数字组成 此处使用AES-128-CBC加密模式，key需要为16位。
     * @param ivParameter    加密参数
     * @return 解密后的字符串
     */
    public static String decrypt(String sSrc, String encodingFormat, String sKey, String ivParameter) {
        try {
            byte[] raw = sKey.getBytes("ASCII");
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
            IvParameterSpec iv = new IvParameterSpec(ivParameter.getBytes());
            cipher.init(Cipher.DECRYPT_MODE, skeySpec, iv);
            byte[] encrypted1 = new BASE64Decoder().decodeBuffer(sSrc);//先用base64解密
            byte[] original = cipher.doFinal(encrypted1);
            return new String(original, encodingFormat);
        } catch (Exception ex) {
            throw new RuntimeException(ex);
        }
    }




}


